//https://leetcode.cn/problems/top-k-frequent-elements/
package codeRandomThoughts.Test347前k个高频元素;

import java.util.HashMap;
import java.util.Map;
import java.util.PriorityQueue;

//二刷
public class Solution2 {
    public int[] topKFrequent(int[] nums, int k) {
        //key是元素 value是元素出现的频率
        HashMap<Integer, Integer> hashMap = new HashMap<>();

        //初始化hashMap
        for (int key : nums) {
            Integer value = hashMap.getOrDefault(key, 0);
            hashMap.put(key, value + 1);
        }

        //建立小顶堆
        //数组中存储元素和对应的出现频率
        PriorityQueue<int[]> priorityQueue = new PriorityQueue<>((pair1, pair2) -> pair1[1] - pair2[1]);//按照频率构造小顶堆
        for (Map.Entry<Integer, Integer> entry : hashMap.entrySet()) {
            if (priorityQueue.size() < k) {
                priorityQueue.add(new int[]{entry.getKey(), entry.getValue()});
            } else {
                //只有当前元素比最小值大才操作
                if (entry.getValue() > priorityQueue.peek()[1]) {
                    priorityQueue.poll();//弹出堆顶元素
                    priorityQueue.add(new int[]{entry.getKey(), entry.getValue()});
                }
            }
        }

        //返回此时优先队列中的元素
        int[] result = new int[k];
        for (int i = k - 1; i >= 0; i--) {
            result[i] = priorityQueue.poll()[0];
        }
        return result;
    }
}
